package _11_整理题目._3_字符串_递归_dfs_bfs_双指针._排列组合._基本_排列组合数;

import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

/**
 * https://leetcode-cn.com/problems/permutation-ii-lcci/
 *
 * 从 n 个 允许有重复元素 的数组中，取出 m 个元素 组成不同的组合 的个数
 *
 * 思路和 无重复n个元素中取值 一样，难点在如何去重！！！
 *      首先，组合数可以抽象为树结构，那么去重存在两个维度，一个是在树枝上，一个是在数层上
 *      题目中，元素在同一个组合内是可以重复的，怎么重复都没事，但两个组合不能相同。
 *      所以我们要去重的是同一树层上的，同一树枝上的都是一个组合里的元素，不用去重。
 *      i>index && nums[i]==nums[i-1] 去重同一层的重复数
 *      https://leetcode-cn.com/problems/subsets-ii/solution/90-zi-ji-iiche-di-li-jie-zi-ji-wen-ti-ru-djmf/
 *
 */
public class _021_组合数_Cnm_允许输入的元素重复 {

    private List<List<Integer>> result = new ArrayList<>();
    private int k;

    public List<List<Integer>> permute(int[] nums, int k) {
        List<Integer> path = new ArrayList<>();
        this.k = k;
        dfs(nums, path, 0);
        return result;
    }

    public void dfs(int[] nums, List<Integer> path, int index) {
        if (path.size() == k) {
            result.add(new ArrayList<>(path));
            return;
        }

        for (int i=index; i < nums.length; i++) {
            if (i > index && nums[i] == nums[i-1]) continue;

            path.add(nums[i]);
            dfs(nums, path, i+1);
            path.remove(path.size()-1);
        }
    }

    @Test
    public void main() {
        int[] nums = {1, 2, 2, 3, 3, 4};
        List<List<Integer>> res = permute(nums, 4);
        res.forEach(System.out::println);
    }
}
